from common.openai_ali import get_client

#定义基础提示词
basic_prompt = "请总结一下段落的主要内容:\n段落：人工智能是研究如何让机器模拟人类智能的科学，主要涉及机器学习、自然语言处理等领域"
#定义优化提示词
optimized_prompt = (
    "以下是一段关于人工智能的描述，请用简洁的语言总结其主要内容，并突出机器学习和自然语言处理两个关键点：\n"
    "段落：人工智能是研究如何让机器模拟人类智能的客户主要涉及机器学习、自然语言处理等领域"
)
# print(basic_prompt)
# print(optimized_prompt)
llm = get_client()
#调用大语言模型生成结果
def generate_response(prompt):
    response = llm.chat.completions.create(
        model="qwen-plus",  # 指定模型
        messages=[
            {'role': 'user', 'content': f'{prompt}'}
        ],
        stream=False  # 非流式输出
    )
    return response.choices[0].message.content

print("基础提示词生成结果：",generate_response(basic_prompt))
print("优化提示词生成结果：",generate_response(optimized_prompt))